package com.guojiang.guopicturebackend.aop;

import com.guojiang.guopicturebackend.annotation.AuthCheck;
import com.guojiang.guopicturebackend.exception.BusinessException;
import com.guojiang.guopicturebackend.exception.ErrorCode;
import com.guojiang.guopicturebackend.model.entity.User;
import com.guojiang.guopicturebackend.model.enums.UserRoleEnum;
import com.guojiang.guopicturebackend.service.UserService;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;

@Aspect
@Component
public class AuthInterceptor {

    @Resource
    private UserService userService;

    /**
     * 执行拦截
     *
     * @param joinPoint
     * @param authCheck
     * @return
     */
    @Around("@annotation(authCheck)")
    public Object doInterceptor(ProceedingJoinPoint joinPoint, AuthCheck authCheck) throws Throwable {
        // RequestContextHolder 是 Spring 框架中用于获取当前请求上下文的工具类。
        // 它提供了一种在不同层次的代码（如服务层、控制器层甚至是在异步任务中）访问当前请求相关信息的机制。
        /**
         * 从RequestContextHolder中 获取当前请求属性（currentRequestAttributes）
         * requestAttributes 本身并非一个特定的类，而是在 Spring 框架中代表请求属性的一种概念
         * 通常由实现了 RequestAttributes 接口的类的实例来承载
         */
        RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
        // ServletRequestAttributes 是 Spring 框架中用于封装 HTTP Servlet 请求相关属性的类。
        // 它实现了 RequestAttributes 接口。
        // 在基于 Servlet 的 Web 应用开发中，ServletRequestAttributes 扮演着重要角色，
        // 使得开发者能够方便地访问和操作与当前 HTTP 请求相关的各种信息。
        /**
         * 将requestAttributes强转为实现该接口的ServletRequestAttributes
         * 并取出request
         */
        HttpServletRequest request = ((ServletRequestAttributes) requestAttributes).getRequest();
        // 获取当前登录用户（这里已经校验是否登录）
        User loginUser = userService.getLoginUser(request);
        // 将字符串转成枚举
        String mustRole = authCheck.mustRole();
        UserRoleEnum mustRoleEnum = UserRoleEnum.getEnumByValue(mustRole);
        // 如果要求具有的角色权限为空，直接放行
        if(mustRoleEnum == null ){
            return joinPoint.proceed();
        }
        // 以下的代码，必须有权限，才能通过
        // 获取当前登录用户的权限枚举类
        UserRoleEnum userRoleEnum = UserRoleEnum.getEnumByValue(loginUser.getUserRole());
        if(userRoleEnum == null){
            throw new BusinessException(ErrorCode.NO_AUTH_ERROR);
        }
        // 要求有管理员权限（mustRoleEnum == ）但是用户没有管理员权限 拒绝
        if(UserRoleEnum.ADMIN.equals(mustRoleEnum) && !UserRoleEnum.ADMIN.equals(userRoleEnum)) {
            throw new BusinessException(ErrorCode.NO_AUTH_ERROR);
        }
        // 通过权限校验，放行
        return joinPoint.proceed();
    }
}
